package br.com.mackinvest.dao;

import static java.lang.System.out;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.TreeMap;

import br.com.mackinvest.constants.Constants;
import br.com.mackinvest.model.Cliente;
import br.com.mackinvest.model.ContaCorrente;

public class ContaCorrenteMySQLDAO implements GenericDAO<ContaCorrente> {
	private PreparedStatement statement;
	private String query;
	private Connection con;
	
	public ContaCorrenteMySQLDAO() {
		con = DBUtil.getConnection();
	}
	
	public void salva(ContaCorrente model) 
	{
		query = "INSERT INTO tbcontacorrente(saldo, id_cliente) VALUES (?,?)";
		try
		{
			statement = con.prepareStatement(query);
			statement.setFloat(1, model.getSaldo());
			statement.setInt(2, model.getCliente().getId());
			
			statement.executeUpdate();
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_INSERT_NAO_REALIZADO);
			out.println(ex);
		}	 
	}
	 
	public void deleta(ContaCorrente model) {
		try
		{
			query = "DELETE FROM tbcontacorrente WHERE id=?";			
			statement = con.prepareStatement(query);
			statement.setInt(1, model.getId());
			statement.executeUpdate();
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_DELETE_NAO_REALIZADO);
			out.println(ex);
		}
			 
	}
	 
	public void atualiza(ContaCorrente model) 
	{
		query = "UPDATE tbcontacorrente SET saldo=?, id_cliente = ? WHERE id = ? ";
		try
		{
			statement = con.prepareStatement(query);
			statement.setFloat(1, model.getSaldo());
			statement.setInt(2, model.getCliente().getId());
			statement.setInt(3, model.getId());
			statement.executeUpdate();
		}
		catch(SQLException ex)
		{
			out.println(Constants.ERRO_UPDATE_NAO_REALIZADO);
			out.println(ex);
		}
	}
	 
	 
	public TreeMap<Integer, ContaCorrente> retornaTodos() 
	{
		query = "SELECT * FROM tbContaCorrente ";
		TreeMap<Integer,ContaCorrente> listaContasRetorno = new TreeMap<Integer, ContaCorrente>();
		ContaCorrente conta_tmp = null;
		try
		{
			statement = con.prepareStatement(query);
			ResultSet rs = statement.executeQuery();
			
			if(rs.next())
			{
				conta_tmp =  new ContaCorrente(rs.getInt("id"),rs.getFloat("saldo"));
				
				Cliente clienteExemplo = new Cliente();
				clienteExemplo.setId( rs.getInt("id_cliente") );
				
				conta_tmp.setCliente( (new ClienteMySQLDAO()).procuraPorExemplo( clienteExemplo ) );
				conta_tmp.setPersistido(true);
				
				listaContasRetorno.put(conta_tmp.getId(),conta_tmp);
			}
		}
		catch(SQLException ex)
		{
			System.out.println(Constants.ERRO_REGISTRO_NAO_ENCONTRADO);
			System.out.println(ex);
		}
		return listaContasRetorno;
	}
	 
	public ContaCorrente procuraPorExemplo(ContaCorrente model) 
	{
		query = "SELECT * FROM tbContaCorrente WHERE id = ?";
		ContaCorrente contaRetorno = null;
		try
		{
			statement = con.prepareStatement(query);
			statement.setInt(1, model.getId());
			statement.execute();
			ResultSet rs = statement.getResultSet();
			if(rs.next()){
				contaRetorno = new ContaCorrente(rs.getInt("id"),rs.getFloat("saldo"));
				
				Cliente clienteExemplo = new Cliente();
				clienteExemplo.setId( rs.getInt("id_cliente") );
				
				contaRetorno.setCliente( (new ClienteMySQLDAO()).procuraPorExemplo( clienteExemplo ) );
				
				contaRetorno.setPersistido(true);
				
			}
		}
		catch(SQLException ex)
		{
			System.out.println(Constants.ERRO_REGISTRO_NAO_ENCONTRADO);
			System.out.println(ex);
		}
		return contaRetorno;
	}
	 
	public ContaCorrente procuraPorCliente(Cliente cliente) {
		query = "SELECT * FROM tbContaCorrente WHERE id_cliente = ?";
		try
		{
			statement = con.prepareStatement(query);
			statement.setInt(1, cliente.getId());
			statement.execute();
			ResultSet rs = statement.getResultSet();
			if(rs.next()){
				ContaCorrente conta = new ContaCorrente(rs.getInt("id"),rs.getFloat("saldo"));
				conta.setCliente( cliente );
				conta.setPersistido(true);
				return conta;
			}
		}
		catch(SQLException ex)
		{
			System.out.println(Constants.ERRO_REGISTRO_NAO_ENCONTRADO);
			System.out.println(ex);
		}
		return null;
	}
}
 
